package com.swak.persistence.datasource;

import javax.sql.DataSource;

import com.swak.reactivex.context.StackThreadContext;

/**
 * 数据源管理器 使用的时候请保证生命周期:配合 DynamicSourceAspect 一起使用
 * 
 * @author liFeng 2014年6月23日
 */
public class DynamicDataSourceContext {

	/**
	 * 使用ThreadLocal维护变量，ThreadLocal为每个使用该变量的线程提供独立的变量副本，
	 * 所以每一个线程都可以独立地改变自己的副本，而不会影响其它线程所对应的副本。
	 */
	private static final StackThreadContext<Object> CONTEXT_HOLDER = new StackThreadContext<>();

	/**
	 * 持有数据源
	 */
	static DynamicDataSource dataSource;

	/**
	 * 设置数据源的变量
	 */
	public static void setDataSourceType(String dsType) {
		CONTEXT_HOLDER.bind(dsType);
	}

	/**
	 * 获得数据源的变量
	 */
	public static Object getDataSourceType() {
		return CONTEXT_HOLDER.get();
	}

	/**
	 * 清空数据源变量
	 */
	public static void clearDataSourceType() {
		CONTEXT_HOLDER.remove();
	}

	/**
	 * 当前的数据源
	 * 
	 * @return
	 */
	public static DataSource determineTargetDataSource() {
		return dataSource.determineTargetDataSource();
	}
}